#!/bin/bash

SHELL_DIR="$(cd $(dirname $0); pwd)"
PROJECT_DIR=$1
SYSTEM_DIR=$2
system_folder_name=`ls *system* -d`
if [[ ! -d ${SYSTEM_DIR} ]]; then
	echo "无SYSTEM目录，爪巴！"
	exit 1
fi
#echo "--->当前项目目工作目录:$PROJECT_DIR" 
#echo "---->插件目录: $SHELL_DIR"
#echo "----->当前项目刷机包的system目录:$SYSTEM_DIR"


lin_dir=${PROJECT_DIR##*/}
if [ ! -d $SHELL_DIR/$lin_dir  ];then
  mkdir $SHELL_DIR/$lin_dir
fi

echo ""
echo ""
echo ""
echo -e "\e[33m     ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ \e[0m"
echo -e "\e[31m      ---->开始尝试去除AVB验证以及data加密<----   \e[0m"
echo -e "\e[33m     ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ \e[0m"
echo ""
fstab=$(sudo find $PROJECT_DIR -name "fstab*")
if [[ $fstab == "" ]];then
	
	echo -e "\e[33m     ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ \e[0m"
	echo -e "\e[31m     >>>>>>>>找不到相关文件,也许没有加密呢>>>>>>>>>  \e[0m"
	echo -e "\e[33m     ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^\e[0m"
	echo ""
	sleep 5
	exit 0
else
	echo -e "\e[31m     >>>>>>  正在去除,请等待....  >>>>>>> \e[0m"
	for file in $fstab; do
		cp $file $SHELL_DIR/$lin_dir
		sed -i 's/,avb.*system//g' $file
		sed -i 's/,avb,/,/g' $file
		sed -i 's/,avb_keys.*key//g' $file
		sed -i 's/fileencryption=ice/encryptable=ice/g' $file
			
	done
fi

for file1 in $fstab; do
	avb_is=$(grep "avb" $file1)
	data_is=$(grep "fileencryption=ice" $file1)
	if [[ $avb_is == "" ]];then
		
		echo -e "\e[33m     ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ \e[0m"
		echo -e "\e[31m     >>>>>>>>>>>>>>  system加密去除成功  >>>>>>>>  \e[0m"
		echo -e "\e[33m     ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ \e[0m"
		echo ""
	else
		
		echo -e "\e[33m     ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ \e[0m"
		echo -e "\e[31m     >>>>>>>>>>   system加密可能去除失败     >>>>> \e[0m"
		echo -e "\e[33m     ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ \e[0m"
		echo ""
	fi

	if [[ $data_is == "" ]];then
		echo -e "\e[33m     ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ \e[0m"
		echo -e "\e[31m     >>>>>>>>>>      data加密去除成功      >>>>>>  \e[0m"
		echo -e "\e[33m     ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ \e[0m"
		echo ""
	else
		
		echo -e "\e[33m     ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ \e[0m"
		echo -e "\e[31m     >>>>>>>>>>>>  data加密可能去除失败   >>>>>>>>  \e[0m"
		echo -e "\e[33m     ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ \e[0m"
		echo ""
	fi
		
done

sleep 5

#chmod -R 0777 $PROJECT_DIR

echo -e "\e[32mDone\e[0m"
